from torch.nn.modules.module import Module
from functions.nnd import NNDFunction
#* new=====================================================
# import torch
# from torch.autograd import Function
#* ================================================================================

class NNDModule(Module):
    def forward(self, input1, input2):
        return NNDFunction()(input1, input2)


